import torch as pt
from python_ai.common.xcommon import *

pt.set_printoptions(edgeitems=100)

sep('3x4')
x = pt.arange(1, 3*4+1).view(3, 4)
print(x)

sep('take')
x2 = x.take(pt.Tensor([[0, 2, 9], [4, 6, 7]]).long())
print(x2)
